package org.example;

import java.io.IOException;
import java.util.Arrays;
import java.util.List;

public class Main {
    public static void main(String[] args) {
        // 检查命令行参数
//        if (args.length < 2) {
//            System.out.println("使用方法: java -jar PDF-adFree.jar <输入PDF路径> <输出PDF路径>");
//            System.out.println("示例: java -jar PDF-adFree.jar input.pdf output.pdf");
//            return;
//        }

//        String inputPath = args[0];
//        String outputPath = args[1];

        String inputPath = "D:\\精选\\10岁前有这6个迹象说明你把孩子教得很好.pdf";
        String outputPath = "D:\\demo\\output10.pdf";

        // 定义结束关键词列表
        List<String> endKeywords = Arrays.asList("【 在看 】", "END", "全文完", "文末点个");

        try {
            PdfAdRemover remover = new PdfAdRemover(endKeywords);
            boolean adsRemoved = remover.removeAds(inputPath, outputPath);

            if (adsRemoved) {
                System.out.println("广告移除成功！已保存至: " + outputPath);
            } else {
                System.out.println("未发现广告内容或结束关键词，已复制原始文件至: " + outputPath);
            }
        } catch (IOException e) {
            System.err.println("处理PDF时发生错误: " + e.getMessage());
            e.printStackTrace();
        }
    }
}